package main

import "fmt"

func convert(s string, numRows int) string {
	// 如果 numRows 为 1，直接返回
	if numRows == 1 {
		return s
	}
	// 定义上下移动的动作
	dir := []int{1, -1}
	ac := 1
	// 控制在哪行
	col := 1
	// 定义数组
	rst := make([][]rune, numRows)
	for i := 0; i < numRows; i++ {
		rst[i] = make([]rune, 0)
	}
	for _, c := range s {
		if col == 1 || col == numRows {
			ac = (ac + 1) % 2
		}
		rst[col-1] = append(rst[col-1], c)
		col += dir[ac]
	}

	strAns := ""
	for _, cArr := range rst {
		strAns += string(cArr)
	}
	return strAns
}

func main() {
	fmt.Println(convert("AB", 1))
}
